home *** CD-ROM | disk | FTP | other *** search
/ PC World 2007 March / PCWorld_2007-03_cd.bin / v cisle / httrack / httrack-3.41-2.exe / {app} / src / htsmms.c < prev    next >
Encoding:
C/C++ Source or Header  |  2006-08-15  |  6.9 KB  |  233 lines
  1. /* ------------------------------------------------------------ */
  2. /*
  3. HTTrack Website Copier, Offline Browser for Windows and Unix
  4. Copyright (C) Xavier Roche and other contributors
  5.  
  6. This program is free software; you can redistribute it and/or
  7. modify it under the terms of the GNU General Public License
  8. as published by the Free Software Foundation; either version 2
  9. of the License, or any later version.
  10.  
  11. This program is distributed in the hope that it will be useful,
  12. but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14. GNU General Public License for more details.
  15.  
  16. You should have received a copy of the GNU General Public License
  17. along with this program; if not, write to the Free Software
  18. Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19.  
  20.  
  21. Important notes:
  22.  
  23. - We hereby ask people using this source NOT to use it in purpose of grabbing
  24. emails addresses, or collecting any other private information on persons.
  25. This would disgrace our work, and spoil the many hours we spent on it.
  26.  
  27.  
  28. Please visit our Website: http://www.httrack.com
  29. */
  30.  
  31.  
  32. /* ------------------------------------------------------------ */
  33. /* File: basic mms protocol manager                             */
  34. /* Author: Xavier Roche                                         */
  35. /*                                                              */
  36. /* The mms routines were written by Nicolas BENOIT,             */
  37. /* based on the work of SDP Multimedia and Major MMS                */
  38. /* Thanks to all of them!                                       */
  39. /* ------------------------------------------------------------ */
  40.  
  41. /* Internal engine bytecode */
  42. #define HTS_INTERNAL_BYTECODE
  43.  
  44. // Gestion protocole mms
  45.  
  46. #include "htsglobal.h"
  47.  
  48. #if HTS_USEMMS
  49.  
  50. #include "htscore.h"
  51.  
  52. #include "htsmms.h"
  53. #include "mmsrip/mms.h"
  54.  
  55. #define FTP_STATUS_READY 1001
  56.  
  57. static int run_launch_mms(MMSDownloadStruct* back);
  58. static void back_launch_mms( void* pP ) {
  59.     MMSDownloadStruct *pStruct = (MMSDownloadStruct*)pP;
  60.   if (pStruct == NULL)
  61.     return ;
  62.  
  63.     /* Initialize */ 
  64.     hts_init();
  65.  
  66.     /* Run */
  67.     run_launch_mms(pStruct);
  68.  
  69.     /* Set as ready */
  70.     {
  71.         lien_back* back = pStruct->pBack;
  72.         back->status=FTP_STATUS_READY;
  73.     }
  74.  
  75.     /* Delete structure */
  76.     free(pP);
  77.  
  78.     /* Uninitialize */
  79.   hts_uninit();
  80.   return ;
  81. }
  82.  
  83. /* download cancelled */
  84. static int stop_mms(lien_back* back) {
  85.     if (back->stop_ftp) {
  86.         strcpybuff(back->r.msg, "Cancelled by User");
  87.         back->r.statuscode = STATUSCODE_INVALID;
  88.         return 1;
  89.     }
  90.     return 0;
  91. }
  92.  
  93. /* Background launch */
  94. void launch_mms(const MMSDownloadStruct* pStruct) {
  95.     MMSDownloadStruct *pCopy = calloc(sizeof(MMSDownloadStruct), 1);
  96.     memcpy(pCopy, pStruct, sizeof(*pCopy));
  97.   hts_newthread(back_launch_mms, (void*) pCopy);
  98. }
  99.  
  100. /* Code mainly written by Nicolas BENOIT */
  101. static int run_launch_mms(MMSDownloadStruct* pStruct) {
  102.     lien_back* back = pStruct->pBack;
  103.     httrackp* opt = pStruct->pOpt;
  104.     /* */
  105.     char url[HTS_URLMAXSIZE*2];
  106.     char catbuff[CATBUFF_SIZE];
  107.     char catbuff2[CATBUFF_SIZE];
  108.   MMS *mms;
  109.   FILE *f;
  110.   ssize_t len_written;
  111.     uint64_t total_len_written;
  112.     int delay = opt->mms_maxtime;
  113.     time_t end = time(NULL) + delay;
  114.     short checkPending = 0;
  115.     ssize_t existingSize = fsize(back->url_sav);
  116.  
  117.     // effacer
  118.     strcpybuff(back->r.msg,"");
  119.     back->status=STATUS_FTP_TRANSFER;
  120.     back->r.statuscode=HTTP_OK;
  121.     back->r.size=0;
  122.  
  123.     /* Create file */
  124.     if (existingSize > 0) {
  125.         /* back->r.out = fileappend(back->url_sav);
  126.         */
  127.         (void) unlink(fconcat(catbuff,back->url_sav, ".old"));
  128.         if (rename(fconcat(catbuff,back->url_sav, ""), fconcat(catbuff2,back->url_sav, ".old")) == 0) {
  129.             checkPending = 1;
  130.         }
  131.         back->r.out = filecreate(&pStruct->pOpt->state.strc, back->url_sav);
  132.     } else {
  133.         back->r.out = filecreate(&pStruct->pOpt->state.strc, back->url_sav);
  134.     }
  135.     if ((f = back->r.out) != NULL) {
  136.         // create mms resource
  137.         strcpybuff(url, back->url_adr);                /* mms:// */
  138.         strcatbuff(url, back->url_fil);
  139.         if ( ( mms = mms_create( url, f, NULL, 0, 1 ) ) != NULL ) {
  140.             if ( mms_connect ( mms ) == 0 ) {
  141.                 if ( mms_handshake ( mms ) == 0 ) {
  142.                     if ( ( len_written = mms_write_stream_header ( mms ) ) != -1 ) {
  143.                         total_len_written = len_written;
  144.                         HTS_STAT.HTS_TOTAL_RECV += len_written;
  145.  
  146.                         /* not modified */
  147.                         if (checkPending) {
  148.                             if (mms->is_live != MMS_LIVE
  149.                                 && mms->expected_file_size == existingSize + 50    /* Why 50 additional bytes declared ?? */
  150.                                 )                // abort download
  151.                             {
  152.                                 fclose(back->r.out);
  153.                                 f = back->r.out = NULL;
  154.                                 if (unlink(fconcat(catbuff, back->url_sav, "")) == 0 
  155.                                     && rename(fconcat(catbuff, back->url_sav, ".old"), fconcat(catbuff2, back->url_sav, "")) == 0) 
  156.                                 {
  157.                                     back->r.notmodified = 1;
  158.                                     back->r.statuscode = HTTP_OK;
  159.                                     strcpybuff(back->r.msg, "Not modified");
  160.                                 } else {
  161.                                     back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  162.                                     strcpybuff(back->r.msg, "Unable to rename previous file (not updated)");
  163.                                 }
  164.                             } else {
  165.                                 (void) unlink(fconcat(catbuff, back->url_sav, ".old"));
  166.                             }
  167.                         }
  168.  
  169.                         /* begin rip */
  170.                         if ( f != NULL && mms_begin_rip ( mms ) == 0 ) {
  171.                             if ( mms->is_live != MMS_LIVE ) {
  172.                                 back->r.totalsize = mms->expected_file_size;
  173.                                 back->r.totalsize -= 50;    /* Why 50 additional bytes declared ?? */
  174.                             } else
  175.                                 back->r.totalsize = -1;
  176.  
  177.                             /* Start download */
  178.                             while ( !stop_mms(back) ) {
  179.                                 len_written = mms_write_stream_data ( mms );
  180.                                 if ( len_written == 0 ) {
  181.                                     break;
  182.                                 } else if ( len_written == -1 ) {
  183.                                     back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  184.                                     strcpybuff(back->r.msg, "Unable to write stream data");
  185.                                     break;
  186.                                 }
  187.  
  188.                                 total_len_written += len_written;
  189.                                 back->r.size = total_len_written;
  190.                                 HTS_STAT.HTS_TOTAL_RECV += len_written;
  191.  
  192.                                 fflush ( f );
  193.  
  194.                                 if ( delay != 0 && end <= time(NULL) ) {
  195.                                     delay = -1;
  196.                                     back->r.statuscode = HTTP_OK;
  197.                                     strcpybuff(back->r.msg, "Download interrupted");
  198.                                     break;
  199.                                 }
  200.                             }        // while
  201.  
  202.                             back->r.statuscode = HTTP_OK;            /* Finished */
  203.                         } else if (f != NULL) {
  204.                             back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  205.                             strcpybuff(back->r.msg, "Can not begin ripping");
  206.                         }
  207.                     } else {
  208.                         back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  209.                         strcpybuff(back->r.msg, "Can not write stream header");
  210.                     }
  211.                 } else {
  212.                     back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  213.                     strcpybuff(back->r.msg, "Can not handshake");
  214.                 }
  215.                 mms_disconnect ( mms );
  216.             } else {
  217.                 back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  218.                 strcpybuff(back->r.msg, "Can not connect");
  219.             }
  220.             mms_destroy ( mms );
  221.         } else {
  222.             back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  223.             strcpybuff(back->r.msg, "Can not create mms resource");
  224.         }
  225.     } else {
  226.         back->r.statuscode = HTTP_INTERNAL_SERVER_ERROR;
  227.         strcpybuff(back->r.msg, "Unable to open local output file");
  228.     }
  229.     return 0;
  230. }
  231.  
  232. #endif
  233.